from openai import OpenAI

# 创建OpenAI客户端
client = OpenAI(
    api_key="sk-117hahihft1nkgc0t6qa5gvhd7q1mvndgcud046o3d4l6hm3",  # 你自己创建创建的Key
    base_url="http://api.aihao123.cn/luomacode-api/open-api/v1"  # 你的base_url
)


def api(content):
    print()

    # 这里是系统提示词
    sysContent = f"请对下面的内容进行分类，并且描述出对应分类的理由。你只需要根据用户的内容输出下面几种类型：bug类型,用户体验问题，用户吐槽." \
                 f"输出格式:[类型]-[问题:{content}]-[分析的理由]"
    response = client.chat.completions.create(
        messages=[
            # 把系统提示词传进来sysContent
            {'role': 'system', 'content': sysContent},
            # 把用户提示词传进来content
            {'role': 'user', 'content': content},
        ],
        # 这是模型
        model='gpt-4',
        stream=True
    )

    for chunk in response:
        print(chunk.choices[0].delta.content, end="", flush=True)


if __name__ == '__main__':
    content = "这个页面不太好看"
    api(content)